#include "udpserver.hpp" // 确保udpserver.hpp与服务端类定义对应
#include <memory>
#include <cstdio>
#include <vector>
#include <iostream>
#include <stdexcept>

void Usage(std::string proc)
{
    // 修正：明确用法（参数为端口号）
    std::cout << "usage: " << proc << " <port> (port >= 1024)\n" << std::endl;
}

// 以下函数（hander、SafeCheck、ExcuteCommand）未被服务端使用，可保留或删除
// std::string hander(const std::string &str)
// {
//     std::string res = "server get a message:";
//     res += str;
//     std::cout << res << std::endl;
//     return res;
// }

// bool SafeCheck(const std::string &cmd)
// {
//     std::vector<std::string> key_word = {
//         "rm", "mv", "cp", "kill", "sudo", 
//         "unlink", "uninstall", "yum", "top", "while"
//     };
//     for (auto &word : key_word)
//     {
//         auto pos = cmd.find(word);
//         if (pos != std::string::npos) 
//             return false;
//     }
//     return true;
// }

// std::string ExcuteCommand(const std::string &cmd)
// {
//     std::cout << "get a request cmd: " << cmd << std::endl;
//     if (!SafeCheck(cmd)) 
//         return "Bad man";

//     FILE *fp = popen(cmd.c_str(), "r");
//     if (nullptr == fp)
//     {
//         perror("popen");
//         return "error";
//     }
//     std::string result;
//     char buffer[4096];
//     while (fgets(buffer, sizeof(buffer), fp) != nullptr)
//     {
//         result += buffer;
//     }
//     pclose(fp);
//     return result;
// }

int main(int argc, char *argv[])
{
    if (argc != 2)
    {
        // 修正1：参数不足时调用Usage，传入argv[0]（程序名），而非argv[2]（空指针）
        Usage(argv[0]);
        exit(1); // 异常退出（用1而非0）
    }

    uint16_t port;
    // 修正2：增加端口号合法性校验（数字+范围）
    try
    {
        port = std::stoi(argv[1]);
        if (port < 1024 || port > 65535)
        {
            std::cerr << "Port must be between 1024 and 65535!" << std::endl;
            exit(1);
        }
    }
    catch (...)
    {
        std::cerr << "Port must be a number!" << std::endl;
        exit(1);
    }

    // 修正3：服务端IP设为0.0.0.0（监听所有网卡，支持局域网/公网访问）
    std::unique_ptr<UdpServer> svr(new UdpServer(port, "0.0.0.0"));
    svr->Init();
    svr->Run();

    return 0;
}